package com.transpot.utils;

import org.springframework.util.ResourceUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.Calendar;
import java.util.UUID;

/**
 * 文件上传工具方法
 */
public class UploadUtil {
    public static String uploadImg(MultipartFile file){
        //获取文件名
        String fileName = file.getOriginalFilename();
        //获取文件后缀名。也可以在这里添加判断语句，规定特定格式的图片才能上传，否则拒绝保存。
        String suffixName = fileName.substring(fileName.lastIndexOf("."));
        //为了避免发生图片替换，这里使用了文件名重新生成
        fileName = UUID.randomUUID() + suffixName;

        Calendar calendar = Calendar.getInstance();
        String prePath = "/" + calendar.get(Calendar.YEAR) + "/" + (calendar.get(Calendar.MONTH) + 1);
        File trueFile = new File("D:/upload" + prePath);
        if (!trueFile.exists()){
            if (!trueFile.mkdirs()){
                return null;
            }
        }
        try {
            file.transferTo(new File("D:/upload" + prePath + "/" + fileName));
        } catch (IOException e) {
            e.printStackTrace();
        }
        return CommonConstant.IMG_PATH + prePath + "/" + fileName;
    }
}